package homework.test1

import scala.io.StdIn

object IsPrime {
  def isPrime(p: Int): Unit = {
    if (p <= 1) {
      println("不是质数")
    } else if (p == 2) {
      println("是质数")
    } else if (p % 2 == 0) {
      println("不是质数")
    } else {
      val num = Math.sqrt(p).toInt
      var isPrime = true  // 假设是质数
      for (i <- 3 to num by 2) {
        if (p % i == 0) {
          isPrime = false
          println("不是质数")
          return  // 提前退出方法
        }
      }
      if (isPrime) {
        println("是质数")
      }
    }
  }

  def main(args: Array[String]): Unit = {
    println("请输入一个大于0的整数：")
    val num = StdIn.readInt()
    isPrime(num)
  }
}